import request from '@/utils/request'

//河流树监测断面
// /tools/segmentRiverTree
export function segmentRiverTree(query = {}) {
  return request({
    url: '/tools/segmentRiverTree',
    method: 'get',
    params: query
  })
}

    export function convertToTree(data) {
      // 创建id到节点的映射表
      const nodeMap = {};

      // 初始化所有节点，保留原有id，添加label和children
      data.forEach((node) => {
        nodeMap[node.id] = {
          ...node, // 保留所有原始字段（包括id）
          label: node.name, // 添加label字段，值为name
          children: [], // 初始化子节点数组
        };
      });

      // 构建树形结构
      const tree = [];
      data.forEach((node) => {
        const currentNode = nodeMap[node.id];
        const parentNode = nodeMap[node.pId];

        if (parentNode) {
          // 存在父节点，添加到父节点的children中
          parentNode.children.push(currentNode);
        } else {
          // 根节点（pId为0）直接加入树
          tree.push(currentNode);
        }
      });

      return tree;
    }